이차원 리스트 전치행렬로 만들기 {python}

• 0014 Python 🐍

count-servers-that-communicate 문제를 풀다가 다른 이의 풀이법을 보고 무릎을 탁 쳤다.

grid = [[1, 1, 0, 0], 
		[0, 0, 1, 0], 
		[0, 0, 1, 0], 
		[0, 0, 0, 1]]

print([list(e) for e in zip(*grid)])
"""
[[1, 0, 0, 0], 
 [1, 0, 0, 0], 
 [0, 1, 1, 0], 
 [0, 0, 0, 1]]
"""